package com.cango.student.classloader.first1;

import java.io.*;
import java.lang.reflect.Method;
import java.util.HashSet;

/**
 * HowswapCL类是自定义的classloader，它有2个成员，basedir和dynaclazns，分别表示需要加载的类的所在目录，和已经加载了的类名，同时实现了ClassLoader的接口loadClass。
 * loadClass根据类名称去加载类实例，它先调用findLoadedClass查找这个类是否已经被加载了，如果没有被加载，
 * 并且dynaclazns里面也没有记录，那么就使用系统加载器加载（getSystemClassLoader().loadClass(name)）；
 * 如果dynaclazns里面有记录，但是还是没有被加载，那么就抛出ClassNotFound异常。
 *
 * 那dynaclazns中的内容是在何时添加的呢？是在初始化classloader的时候，也就是初始化HowswapCL类的时候，
 * 初始化过程中，先对每个basedir路径下需要加载的类文件调用loadDirectly做加载，然后将类名加入到dynaclazns中。在loadDirectly做加载的时候，
 * 先通过basedir路径和类名拼出class文件的路径，然后将class文件以二进制形式读入到对象raw中，最后根据类名和raw中二进制信息，调用defineClass加载这个类。
 *
 * @program: student
 * @description:
 * @author: xie jiefeng
 * @create: 2019-05-15 10:54
 */
public class HowswapCL extends ClassLoader {
    private String basedir; // 需要该类加载器直接加载的类文件的基目录
    private HashSet<String> dynaclazns; // 需要由该类加载器直接加载的类名

    public HowswapCL(String basedir, String[] clazns) {
        super(null); // 指定父类加载器为 null
        this.basedir = basedir;
        dynaclazns = new HashSet<String>();
        loadClassByMe(clazns);
    }

    private void loadClassByMe(String[] clazns) {
        for (int i = 0; i < clazns.length; i++) {
            loadDirectly(clazns[i]);
            dynaclazns.add(clazns[i]);
        }
    }

    private Class<?> loadDirectly(String name) {
        Class<?> cls = null;
        StringBuffer sb = new StringBuffer(basedir);
        String classname = name.replace('.', File.separatorChar) + ".class";
        sb.append(File.separator + classname);
        File classF = new File(sb.toString());
        try {
            cls = instantiateClass(name, new FileInputStream(classF),
                    classF.length());
        } catch (FileNotFoundException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        return cls;
    }

    private Class<?> instantiateClass(String name, InputStream fin, long len) {
        byte[] raw = new byte[(int) len];
        try {
            fin.read(raw);
            fin.close();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        return defineClass(name, raw, 0, raw.length);
    }

    @Override
    protected Class<?> loadClass(String name, boolean resolve)
            throws ClassNotFoundException {
        Class<?> cls = null;
        cls = findLoadedClass(name);
        if (!this.dynaclazns.contains(name) && cls == null) {
            cls = getSystemClassLoader().loadClass(name);
        }

        if (cls == null) {
            throw new ClassNotFoundException(name);
        }

        if (resolve) {
            resolveClass(cls);
        }

        return cls;
    }
}

/*
 * 每隔2s运行一次，不断加载class
 *
 *
 */
class Multirun implements Runnable {
    @Override
    public void run() {
        try {
            while (true) {
                // 每次都创建出一个新的类加载器
                // class需要放在自己package名字的文件夹下
//                String url = System.getProperty("user.dir") + "/Users/xiejeifeng/Learn";// "/bin/problem1/GetInfo.class";
                String url = "/Users/xiejiefeng/Learn";// "/bin/problem1/GetInfo.class";
                HowswapCL cl = new HowswapCL(url,
                        new String[]{"problem1.GetInfo"});
                Class<?> cls = cl.loadClass("problem1.GetInfo");
                Object foo = cls.newInstance();
                // 被调用函数的参数
                Method m = foo.getClass().getMethod("Output", new Class[]{});
                m.invoke(foo, new Object[]{});
                Thread.sleep(2000);
            }
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }
}
